import streamlit as st
import re
import data.data
import time

st.set_page_config(
    page_title="个人助手登录界面",
    page_icon="💖✌️"
)
st.title('个人助手登录界面')
username = st.text_input("请输入用户名")
password = st.text_input("请输入密码",type='password')
loginFlag = st.button("登录")
registerFlag = st.button("没有账号，点击注册")

if loginFlag:
    if username and password:
        if re.match('^(13|15|17|18|19)[0-9]{9}$', username):
            if len(password) >= 8:
                if data.data.query_user(username):
                     res = data.data.query_user_password(username)
                     if res:
                        if res["password"] == password:
                            st.success("登录成功")
                            time.sleep(2)
                            st.switch_page("pages/chatbot.py")
                        else:
                            st.error('用户密码错误')
                     else:
                         st.error('用户不存在')
                else:
                    st.error("账号或者密码不正确")
            else:
                st.error("密码长度不足8位")
        else:
            st.error('您的手机号码格式不正确')
    else:
        st.error("请完善信息")

if registerFlag:
    st.switch_page("pages/register.py")

